from utils import *

def show_menu():
    print("\n=== 学生成绩管理系统 ===")
    print("1. 添加学生成绩")
    print("2. 查询学生成绩")
    print("3. 查看班级统计")
    print("4. 退出系统")

def main():
    while True:
        show_menu()
        choice = input("请输入选项：")
        if choice == "1":
            add_student()
        elif choice == "2":
            search_student()
        elif choice == "3":
            calculate_statistics()
        elif choice == "4":
            print("感谢使用！")
            break
        else:
            print("输入错误，请重新输入")

if __name__ == "__main__":
    main()